import pygame
import time
from pygame.locals import *
import random


class Me(pygame.sprite.Sprite):
    def __init__(self, screen):
        # 精灵类的初始化方法 必须调用
        pygame.sprite.Sprite.__init__(self)

        # 创建一个图片当飞机
        self.player = pygame.image.load("me1.png")

        # 根据图片image获取矩形对象
        self.rect = self.player.get_rect()  # rect:矩形
        self.rect.topleft = [190, 600]

        # 飞机速度
        self.speed = 10

        # 记录当前的窗口对象
        self.screen = screen

        # 装子弹的列表
        self.bullets = pygame.sprite.Group()

    def key_control(self):
        # 监听键盘事件
        key_pressed = pygame.key.get_pressed()

        if key_pressed[K_w] or key_pressed[K_UP]:
            self.rect.top -= self.speed
        if key_pressed[K_s] or key_pressed[K_DOWN]:
            self.rect.bottom += self.speed
        if key_pressed[K_a] or key_pressed[K_LEFT]:
            self.rect.left -= self.speed
        if key_pressed[K_d] or key_pressed[K_RIGHT]:
            self.rect.right += self.speed
        if key_pressed[K_SPACE]:
            # 按下空格发射子弹
            bullet = Bullet(self.screen, self.rect.left, self.rect.top)
            # 把子弹放到列表里
            self.bullets.add(bullet)

    def update(self):
        self.key_control()
        self.display()

    def display(self):
        # 将飞机图片贴到窗口中
        self.screen.blit(self.player, self.rect)
        # 更新子弹坐标
        self.bullets.update()

        # 把所有子弹添加到屏幕
        self.bullets.draw(self.screen)


class Enemy(pygame.sprite.Sprite):
    def __init__(self, screen):
        # 精灵类的初始化方法 必须调用
        pygame.sprite.Sprite.__init__(self)

        # 创建一个图片当飞机
        self.player = pygame.image.load("enemy1.png")

        # 根据图片image获取矩形对象
        self.rect = self.player.get_rect()  # rect:矩形
        self.rect.topleft = [0, 0]

        # 飞机速度
        self.speed = 5

        # 记录当前的窗口对象
        self.screen = screen

        # 装子弹的列表
        self.bullets = pygame.sprite.Group()

        # 敌机移动方向
        self.direction = 'right'

    def display(self):
        # 将飞机图片贴到窗口中
        self.screen.blit(self.player, self.rect)

        # 更新子弹坐标
        self.bullets.update()

        # 把所有子弹添加到屏幕
        self.bullets.draw(self.screen)

    def update(self):
        self.auto_move()
        self.auto_fire()
        self.display()

    def auto_move(self):
        # 敌机左右移动
        if self.direction == 'right':
            self.rect.right += self.speed
        elif self.direction == 'left':
            self.rect.right -= self.speed

        # 触碰边界方向相反
        if self.rect.right > 480:
            self.direction = 'left'
        elif self.rect.right < 57:
            self.direction = 'right'

    def auto_fire(self):
        """自动开火 创建子弹对象 添加进列表"""
        random_num = random.randint(1, 20)
        if random_num == 8:
            bullet = EnemyBullet(self.screen, self.rect.left, self.rect.right)
            self.bullets.add(bullet)


# 子弹类
# 属性
class Bullet(pygame.sprite.Sprite):
    def __init__(self, screen, x, y):
        # 精灵类初始化
        pygame.sprite.Sprite.__init__(self)

        # 创建图片
        self.image = pygame.image.load('bullet1.png')

        # 获取矩形对象
        self.rect = self.image.get_rect()
        self.rect.topleft = [x + 100 / 2, y]

        # 窗口
        self.screen = screen
        # 速度
        self.speed = 10

    def update(self):
        # 修改子弹坐标
        self.rect.top -= self.speed
        # 如果子弹移出屏幕上方 销毁子弹
        if self.rect.top < -20:
            self.kill()


# 敌方子弹类
# 属性
class EnemyBullet(pygame.sprite.Sprite):
    def __init__(self, screen, x, y):  # 精灵类初始化
        pygame.sprite.Sprite.__init__(self)

        # 创建图片
        self.image = pygame.image.load('bullet2.png')

        # 获取矩形对象
        self.rect = self.image.get_rect()
        self.rect.topleft = [x + 58 / 2 - 6 / 2, 43]

        # 窗口
        self.screen = screen
        # 速度
        self.speed = 5

    def update(self):
        # 修改子弹坐标
        self.rect.top += self.speed
        # 如果子弹移出屏幕上方 销毁子弹
        if self.rect.top > 720:
            self.kill()


class GameSound(object):
    def __init__(self):
        pygame.mixer.init()  # 音乐模块初始化
        pygame.mixer.music.load('Toby Fox - Dummy!.mp3')
        pygame.mixer.music.set_volume(0.5)  # 声音大小

        self.__bomb = pygame.mixer.Sound('近藤浩治 - プレイヤーダウン (スーパーマリオブラザーズ効果音).mp3')

    def playBackgroundMusic(self):
        pygame.mixer.music.play(-1) #开始播放音乐

    def playBombSound(self):
        pygame.mixer.Sound.play(self.__bomb)

    def playBackgroundMusic(self):
        pygame.mixer.music.play(-1)  # 开始播放音乐


class Bomb(object):
    def __init__(self, screen, type):
        self.screen = screen
        if type == "enemy":
            # 加载爆炸资源
            self.mImage = [pygame.image.load("enemy1_down" + str(v) + ".png") for v in range(1, 5)]
        else:
            self.mImage = [pygame.image.load("me_destroy_" + str(v) + ".png") for v in range(1, 5)]
        # 设计当前爆炸播放索引
        self.mIndex = 0
        # 爆炸设置
        self.mPos = [0, 0]
        # 是否可见
        self.mVisible = False

    #爆炸位置设置
    def action(self, rect):
        # 触发爆炸的方法draw
        self.draw()
        # 爆炸的坐标
        self.mPos[0] = rect.left
        self.mPos[1] = rect.top
        # 打开爆炸的开关
        self.mVisible = True

    #爆炸图像切换的方法
    def draw(self):
        #如果是关闭爆炸状态，则不执行后面代码，直接返回
        if not self.mVisible:
            return
        #绘制当前编号图片
        self.screen.blit(self.mImage[self.mIndex],(self.mPos[0],self.mPos[1]))
        #图片编号+1
        self.mIndex += 1
        #如果下标到最后一张图片了，则表示爆炸结束
        if self.mIndex >= len(self.mImage):
            self.mIndex = 0
            self.mVisible = False
        #while self.mVisible:
        #    for image in self.mImage:
        #        self.screen.blit(image,self.mPos)
        #        pygame.display.flip()
        #        self.mVisible = False

class Manager(object):
    def __init__(self):
        #创建窗口
        self.screen = pygame.display.set_mode((480,700),0,32)
        #创建背景图片
        self.background = pygame.image.load("background.png")
        #初始化一个装玩家精灵的group
        self.players = pygame.sprite.Group()
        #初始化一个装敌机精灵的group
        self.enemys = pygame.sprite.Group()
        #初始化一个玩家爆炸的对象
        self.player_bomb = Bomb(self.screen,'player')
        #初始化一个敌机爆炸的对象
        self.enemy_bomb = Bomb(self.screen,'enemy')
        #初始化一个声音播放的对象
        self.sound = GameSound()

    def exit(self):
        print('退出')
        pygame.quit()
        exit()

    def new_player(self):
        #创建飞机对象 添加到玩家组
        player = Me(self.screen)
        self.players.add(player)

    def new_enemy(self):
        #创建飞机对象 添加到玩家组
        enemy = Enemy(self.screen)
        self.enemys.add(enemy)

    #绘制文字
    def drawText(self,text,x,y,textHeigh=30,fontColor=(255,0,0),backgroundColor=None):
        #通过字体文件获取文字对象
        font_obj = pygame.font.Font('STCAIYUN.TTF',textHeigh)
        #配置要显示的文字
        text_obj = font_obj.render(text,True,fontColor,backgroundColor)
        #获取要显示的对象的rect
        text_rect = text_obj.get_rect()
        #设置显示对象的坐标
        text_rect.topleft = (x,y)
        #绘制字到指定区域
        self.screen.blit(text_obj,text_rect)


    def main(self):
        #播放背景音乐
        self.sound.playBackgroundMusic()
        #创建一个玩家
        self.new_player()
        #创建敌机
        self.new_enemy()
        while True:
            #把背景图片贴到窗口
            self.screen.blit(self.background, (0, 0))
            # 遍历所有事件
            for event in pygame.event.get():
                # 判断事件类型
                if event.type == QUIT:
                    # python程序退出
                    self.exit()

            #调用爆炸的对象
            self.player_bomb.draw()
            self.enemy_bomb.draw()

            #判断碰撞
            iscollide = pygame.sprite.groupcollide(self.players,self.enemys,True,True)

            if iscollide:
                items = list(iscollide.items())[0]
                print(items)
                x = items[0]
                y = items[1][0]
                #玩家爆炸图片
                self.player_bomb.action(x.rect)
                #敌机爆炸图片
                self.enemy_bomb.action(y.rect)
                #爆炸的声音
                self.sound.playBombSound()

            #玩家飞机和子弹的显示
            self.players.update()
            #敌机和子弹的显示
            self.enemys.update()

            #刷新窗口内容
            pygame.display.update()
            time.sleep(0.02)




if __name__ == '__main__':
    manager = Manager()
    manager.main()
